"""
SQL 综合案例，读取文件，写入MySQL数据库中
"""
from file_define import *
from data_define import *
from pyecharts.charts import Bar
from pyecharts.options import *
from pyecharts.globals import ThemeType
from pymysql import Connection
text_file_reader = TextFileReader("2011年1月销售数据.txt")
json_file_reader = JsonFileReader("2011年2月销售数据JSON.txt")

jan_data:list[Record] = text_file_reader.read_data()
feb_data:list[Record] = json_file_reader.read_data()
# 将两个月的数据整合到一个列表
all_data = jan_data + feb_data

# 构建MySQL链接对象
conn = Connection(
    host="localhost",
    port=3306,
    user="root",
    password="hzh3589153651",
    autocommit=True
)
# 获得游标对象
cursor = conn.cursor()
# 选择数据库
conn.select_db('py_sql')
# 组织SQL语言
for record in all_data:
    sql = f"insert into orders(order_date, order_id, money,province)"\
          f"values('{record.data}','{record.order_id}',{record.money},'{record.province}')"
    # 执行sql语句
    cursor.execute(sql)

# 关闭sql链接
conn.close()